modification training model in one file